package main

import (
	"github.com/zeromicro/go-zero/core/conf"
	"gorm.io/driver/mysql"
	"gorm.io/gen"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
	"gozero-demo/internal/config"
	"strings"
)

func intTypeHandler(columnType gorm.ColumnType) (dataType string) {
	return "int64"
}

// 时间类型统一处理, 如果数据表的时间类型的字段可以为空时，使用指针类型
func timeTypeHandler(columnType gorm.ColumnType) (dataType string) {
	isNullable, _ := columnType.Nullable()
	if isNullable {
		return "*time.Time"
	}
	return "time.Time"
}

// GEN 自动生成 GORM 模型结构体文件及使用示例
// 安装: go get -u gorm.io/gen@v0.3.16
// 更多参考: https://gorm.io/zh_CN/gen | https://gorm.io/gen/database_to_structs.html
// 更多参考: https://segmentfault.com/a/1190000042502370
func main() {
	// 初始化go-zero的配置
	var c config.Config
	configFile := config.GetConfigFile("") //调用自定义的GetConfigFile方法，读取当前配置的env信息
	conf.MustLoad(configFile, &c)

	// 连接数据库
	db, _ := gorm.Open(mysql.Open(c.Mysql.DataSource), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   "",   // 表名前缀
			SingularTable: true, // 使用单数表名，启用该选项，会区分 user 和 users 表为两个不同的数据表
		},
	})

	// 生成实例
	g := gen.NewGenerator(gen.Config{
		// 生成的model文件的路径
		OutPath: "./internal/model/dao/query",

		// WithDefaultQuery 生成默认查询结构体(作为全局变量使用), 即`Q`结构体和其字段(各表模型)
		// WithoutContext 生成没有context调用限制的代码供查询
		// WithQueryInterface 生成interface形式的查询代码(可导出), 如`Where()`方法返回的就是一个可导出的接口类型
		Mode: gen.WithDefaultQuery | gen.WithQueryInterface,

		// 表字段可为 null 值时, 对应结体字段使用指针类型
		//FieldNullable: true, // generate pointer when field is nullable

		// 表字段默认值与模型结构体字段零值不一致的字段, 在插入数据时需要赋值该字段值为零值的, 结构体字段须是指针类型才能成功, 即`FieldCoverable:true`配置下生成的结构体字段.
		// 因为在插入时遇到字段为零值的会被GORM赋予默认值. 如字段`age`表默认值为10, 即使你显式设置为0最后也会被GORM设为10提交.
		// 如果该字段没有上面提到的插入时赋零值的特殊需要, 则字段为非指针类型使用起来会比较方便.
		FieldCoverable: false, // generate pointer when field has default value, to fix problem zero value cannot be assign: https://gorm.io/docs/create.html#Default-Values

		// 模型结构体字段的数字类型的符号表示是否与表字段的一致, `false`指示都用有符号类型
		FieldSignable: false, // detect integer field's unsigned type, adjust generated data type
		// 生成 gorm 标签的字段索引属性
		FieldWithIndexTag: false, // generate with gorm index tag
		// 生成 gorm 标签的字段类型属性
		FieldWithTypeTag: true, // generate with gorm column type tag
	})
	// 设置目标 db
	g.UseDB(db)

	// 自定义字段的数据类型
	dataMap := map[string]func(columnType gorm.ColumnType) (dataType string){
		// 统一数字类型为int64,兼容protobuf
		"tinyint":   intTypeHandler,
		"smallint":  intTypeHandler,
		"mediumint": intTypeHandler,
		"bigint":    intTypeHandler,
		"int":       intTypeHandler,

		// 时间类型统一处理
		"datetime": timeTypeHandler,
	}
	// 要先于`ApplyBasic`执行
	g.WithDataTypeMap(dataMap)

	//=================== 生成全部数据表的model ===================//
	// 自定义模型结体字段的标签
	// 将特定字段名的 json 标签加上`string`属性,即 MarshalJSON 时该字段由数字类型转成字符串类型
	jsonField := gen.FieldJSONTagWithNS(func(columnName string) (tagContent string) {
		//toStringField := `balance, `
		toStringField := ``
		if strings.Contains(toStringField, columnName) {
			return columnName + ",string"
		}
		return columnName
	})
	// 将非默认字段名的字段定义为自动时间戳和软删除字段;
	// 自动时间戳默认字段名为:`updated_at`、`created_at, 表字段数据类型为: INT 或 DATETIME
	// 软删除默认字段名为:`deleted_at`, 表字段数据类型为: DATETIME
	//autoUpdateTimeField := gen.FieldGORMTag("update_time", "column:update_time;type:int unsigned;autoUpdateTime")
	//autoCreateTimeField := gen.FieldGORMTag("create_time", "column:create_time;type:int unsigned;autoCreateTime")
	// 模型自定义选项组
	//fieldOpts := []gen.ModelOpt{jsonField, autoCreateTimeField, autoUpdateTimeField}
	fieldOpts := []gen.ModelOpt{jsonField}

	// 创建模型的结构体,生成文件在 model 目录; 先创建的结果会被后面创建的覆盖
	// 创建全部模型文件, 并覆盖前面创建的同名模型
	allModel := g.GenerateAllTable(fieldOpts...)

	// 创建模型的方法,生成文件在 query 目录; 先创建结果不会被后创建的覆盖
	g.ApplyBasic(allModel...)

	//=================== 生成指定数据表的model ===================//
	//有时候其他小伙伴改动了某个表，不能随着当前版本上线，就需要指定部分数据表
	/*
		g.ApplyBasic(
			g.GenerateModel("users"),
			g.GenerateModel("orders"),
			g.GenerateModel("goods"),
		)
	*/

	g.Execute()
}
